"""Utility tools for the wealth advisor (lightweight, test-friendly)."""

def query_shanghai_index(_: str = "") -> str:
    """上证指数实时查询工具（模拟版）——返回固定的模拟行情数据。

    Kept simple to avoid external API dependency during tests.
    """
    name = "上证指数"
    price = "3125.62"
    change = "6.32"
    pct = "0.20"
    return f"{name} 当前点位: {price}，涨跌: {change}，涨跌幅: {pct}%（模拟数据）"
# tools.py

"""
工具模块 - 财富管理投顾AI助手

该模块包含智能体可以使用的工具和实用函数，用于执行特定任务，例如查询市场数据。
"""

def query_market_data(market: str) -> dict:
    """查询市场数据的工具函数（示例）"""
    # 这里可以实现实际的市场数据查询逻辑
    return {
        "market": market,
        "price": 100.0,  # 示例价格
        "change": 1.5,   # 示例涨跌
        "percentage_change": 1.5  # 示例涨跌幅
    }

def calculate_investment_return(investment: float, rate_of_return: float, years: int) -> float:
    """计算投资回报的工具函数"""
    return investment * ((1 + rate_of_return) ** years)

# 其他工具函数可以在此处添加